import argparse
from Trainer import Trainer
from Trainer2 import Trainer as T2
from SVM import SVMModel
from KNN import KNNModel
from Ensemble import EnsembleModel
from RandomForest import RandomForest
parser = argparse.ArgumentParser("决策树")
parser.add_argument("--train", action="store_true", help="是否开始训练")
parser.add_argument("--data_path", type=str, help="数据集所在路径", default="./data/")
args = parser.parse_args()


if __name__ == "__main__":

    # 第一次预测
    # trainer = Trainer(args)
    # trainer.train_dt()
    # # trainer.test()
    # trainer.test2()


    # # 互补日期预测
    # trainer = T2(args)
    # trainer.train_dt()
    # # trainer.test()
    # trainer.test2()

    # # 互补日期预测，ensemble
    # trainer = EnsembleModel(args)
    # trainer.train_dt()
    # # trainer.test()
    # trainer.test2()

    # # 互补日期预测，svm
    # trainer = SVMModel(args)
    # trainer.train_dt()
    # # trainer.test()
    # trainer.test2()

     # 互补日期预测，svm
    # trainer = KNNModel(args)
    # trainer.train_dt()
    # # trainer.test()
    # trainer.test2()

    trainer = RandomForest(args)
    trainer.train_dt()
    # trainer.test()
    trainer.test2()